package style.Leetcode.初级算法.tree.tree_20220919_1_二叉树的层序遍历;

import style.Leetcode.初级算法.tree.TreeNode;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * 给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。
 * <p>
 * 输入：root = [3,9,20,null,null,15,7]
 * 输出：[[3],[9,20],[15,7]]
 * <p>
 * 输入：root = [1]
 * 输出：[[1]]
 * <p>
 * 输入：root = []
 * 输出：[]
 */
public class Solution {
    /**
     * BFS
     */
    public List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null) return new ArrayList<>();

        // 队列
        Queue<TreeNode> queue = new LinkedList<>();
        List<List<Integer>> resList = new ArrayList<>();

        // 根节点入队
        queue.add(root);
        // 如果不为空就继续循环
        while (!queue.isEmpty()) {
            // BFS 打印， levelNum 表示的是每层的节点数
            int levelNum = queue.size();
            // subList存储的是每层的结点值
            List<Integer> subList = new ArrayList<>();
            for (int i = 0; i < levelNum; i++) {
                // 出队
                TreeNode node = queue.poll();
                subList.add(node.val);

                // 左右子节点如果不为空就加入到队列中
                if (node.left != null) queue.add(node.left);
                if (node.right != null) queue.add(node.right);
            }
            resList.add(subList);
        }
        return resList;
    }


    public static void main(String[] args) {
        TreeNode treeNode = new TreeNode(3);
        treeNode.left = new TreeNode(9);
        treeNode.left.left = null;
        treeNode.left.right = null;
        treeNode.right = new TreeNode(20);
        treeNode.right.left = new TreeNode(15);
        treeNode.right.left.left = null;
        treeNode.right.left.right = null;
        treeNode.right.right = new TreeNode(7);
        treeNode.right.right.left = null;
        treeNode.right.right.right = null;

        Solution solution = new Solution();
        List<List<Integer>> lists = solution.levelOrder(treeNode);
    }
}
